package cn.cnic.security.nssap.stearm.etl.dao;

import cn.cnic.security.nssap.stearm.etl.entity.sandboxEntity;

/**
 * Created by xiaojianping
 * Data:2021/6/16
 * Time:16:01
 */
public class sandboxDao extends DBUtils {
  public int add(sandboxEntity entity){
    try{
        String sql = "insert into file_threat values(DEFAULT,?,?,?,?,?,?,?,?,?,?,?,?,?,?)";
        Object[] o=new Object[14];
        if(entity.getHazard_level().isEmpty()){
           o[0] = null;
        }
        else{
           o[0]=entity.getHazard_level();
        }

        if(entity.getWhite_id().isEmpty()){
           o[1] = null;
        }
        else{
           o[1]=entity.getWhite_id();
        }

        if(entity.getFile_name().isEmpty()){
           o[2] = null;
        }
        else{
           o[2]=entity.getFile_name();
        }

        if(entity.getRepeat_count().isEmpty()){
           o[3] = null;
        }
        else{
           o[3]=entity.getRepeat_count();
        }

        if(entity.getAlarm_sip().isEmpty()){
           o[4] = null;
        }
        else{
           o[4]=entity.getAlarm_sip();
        }

        if(entity.getGroup_id().isEmpty()){
           o[5] = null;
        }
        else{
           o[5]=entity.getGroup_id();
        }

        if(entity.getFile_type().isEmpty()){
           o[6] = null;
        }
        else{
           o[6]=entity.getFile_type();
        }

        if(entity.getProto().isEmpty()){
           o[7] =null;
        }
        else{
           o[7]=entity.getProto();
        }

        if(entity.getAttack_sip().isEmpty()){
           o[8] = null;
        }
        else{
           o[8]=entity.getAttack_sip();
        }

        if(entity.getThreat_type().isEmpty()){
           o[9] = null;
        }
        else{
           o[9]=entity.getThreat_type();
        }

        if(entity.getHost_state().isEmpty()){
           o[10] = null;
        }
        else{
           o[10]=entity.getHost_state();
        }

        if(entity.getAccess_time().isEmpty()){
           o[11] = null;
        }
        else{
           o[11]=entity.getAccess_time();
        }

        if(entity.getMd5().isEmpty()){
           o[12] = null;
        }
        else{
           o[12]=entity.getMd5();
        }

        if(entity.getStatus().isEmpty()){
           o[13] = null;
        }
        else{
           o[13]=entity.getStatus();
        }
        //            Object[] o = {a.getFirst_access_time(),a.getHazard_level(),a.getWhite_id(),a.getType_chain(),a.getAttack(),a.getRepeat_count(),
        //                    a.getUri(),a.getX_forwarded_for(),a.getRsp_status(),a.getAlarm_source(),a.getAlarm_sip(),a.getThreat_name(),a.getGroup_id(),
        //                    a.getAttack_org(),a.getAttack_sip(), a.getHost(),a.getHost_state(),a.getIoc(),a.getAccess_time(),a.getAttack_chain()};
        row = this.getzsg(sql, o);
        if(row > 0){
           System.out.println("插入成功");
           return 1;
        }
        else{
           System.out.println("插入失败");
           return 0;
        }
    }catch(Exception e){
        e.printStackTrace();
        return 0;
    }
  }
}
